from db.mysql import MysqlClientService,MySqlServerService
from db.influx import InfluxdbClientService,InfluxServerService

client_db_dict = dict(
    mysql=MysqlClientService,
    influx=InfluxdbClientService
)

server_db_dict = dict(
    mysql=MySqlServerService,
    influx=InfluxServerService
)


def get_migrate_client_factory(db_type="mysql"):
    return client_db_dict[db_type]


def get_migrate_server_factory(db_type="mysql"):
    return server_db_dict[db_type]